package org.bluedream.core.config.shiro;

import org.apache.shiro.session.Session;
import org.apache.shiro.session.SessionListener;

import java.util.concurrent.atomic.AtomicInteger;

/**
 * Shiro Session 监听管理
 */
public class CustomShiroSessionListener implements SessionListener {
    /**
     * 统计在线人数
     * JUC包 线程安全自增
     */
    private final AtomicInteger sessionCount = new AtomicInteger(0);

    /**
     * 会话创建时触发
     * @param session
     */
    @Override
    public void onStart(Session session) {
        // 会话创建时，在线人数 + 1
        sessionCount.incrementAndGet();
    }

    /**
     * 退出会话时
     * @param session
     */
    @Override
    public void onStop(Session session) {
        // 会话退出时， 在线人数 -1
        sessionCount.decrementAndGet();
    }

    /**
     * 会话过期时
     * @param session
     */
    @Override
    public void onExpiration(Session session) {
        // 在线人数 -1
        sessionCount.decrementAndGet();
    }

    /**
     * 获取 在线人数
     * @return
     */
    public AtomicInteger getSessionCount(){
        return sessionCount;
    }
}
